@echo off
setlocal enabledelayedexpansion

REM 初始化信息
echo 正在启动增强型数据清洗工具...
echo.

REM 检查Python安装
python --version >nul 2>&1
if %errorlevel% neq 0 (
    echo [错误] 未找到Python。请安装Python 3.7或更高版本后再试。
    goto end
)

REM 检查Python 3.12兼容性
for /f "tokens=2" %%a in ('python --version') do set pyver=%%a
echo 检测到Python版本: %pyver%
echo %pyver% | findstr /r "3\.12\.*" >nul
if %errorlevel% equ 0 (
    echo [警告] 检测到Python 3.12。某些NLP库可能存在兼容性问题。
    echo 是否继续？(Y/N)
    set /p choice=
    if /i not "!choice!"=="Y" goto end
)

REM 安装依赖
echo 正在检查NLP依赖...
python install_nlp_dependencies.py
if %errorlevel% neq 0 (
    echo [警告] NLP依赖可能未完全安装，部分功能可能不可用。
)
echo.

REM 创建虚拟环境目录
if not exist "env" (
    mkdir env
)

REM 检查增强处理器是否可用
set enhanced_available=0
if exist "backend\data_clean\enhanced_processor.py" (
    set enhanced_available=1
    echo [信息] 检测到增强型处理器可用。将使用增强型处理进行数据清洗。
) else (
    echo [信息] 未检测到增强型处理器，将使用基础处理进行数据清洗。
)
echo.

REM 用户菜单
:menu
echo 请选择操作:
echo 1. 运行数据清洗
echo 2. 测试样本
echo 3. 退出
echo.
set /p choice=请输入选择(1-3): 

if "%choice%"=="1" (
    echo 开始数据清洗...
    if %enhanced_available%==1 (
        python main.py --processor enhanced --mode clean
    ) else (
        python main.py --processor basic --mode clean
    )
    echo 数据清洗完成。
    pause
    goto menu
)

if "%choice%"=="2" (
    echo 运行测试样本...
    python main.py --mode test
    echo 测试样本运行完成。
    pause
    goto menu
)

if "%choice%"=="3" (
    goto end
)

echo 无效选择，请重试。
goto menu

:end
echo 感谢使用增强型数据清洗工具。
pause 